package com.tanhua.dubbo.server.service;

import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.tanhua.dubbo.server.config.HuanXinConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.retry.annotation.Retryable;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class TokenService {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    private static final String REDIS_KEY = "HX_TOKEN";

    @Autowired
    private HuanXinConfig huanXinConfig;

    /**
     * 获取token，先从redis中获取，如果没有，再去环信接口获取
     *
     * @return
     */
    public String getToken() {
        // 先从redis中获取
        String token = this.redisTemplate.opsForValue().get(REDIS_KEY);
        if (StrUtil.isNotEmpty(token)) {
            return token;
        }

        //访问环信接口获取token
        return this.refreshToken();
    }

    /**
     * 刷新token，请求环信接口，将token存储到redis中
     * 重试机制 可以自己写
     * @return
     */
    //@Retryable
    public String refreshToken() {
        // 重试机制  递归  控制遍历的次数   3 - 5  1 * 2000
        // a1.xxx/{org_name}/{app_name}/token
        String targetUrl = this.huanXinConfig.getUrl() +
                this.huanXinConfig.getOrgName() + "/" +
                this.huanXinConfig.getAppName() + "/token";

        // 拼凑请求参数
        Map<String, Object> param = new HashMap<>();
        param.put("grant_type", "client_credentials");
        param.put("client_id", this.huanXinConfig.getClientId());
        param.put("client_secret", this.huanXinConfig.getClientSecret());

        // 发送post请求
        // HttpRequest Hutool包中的工具
        // RestTemplate
        HttpResponse response = HttpRequest.post(targetUrl)
                .body(JSONUtil.toJsonStr(param))
                .timeout(20000) //请求超时时间
                .execute(); // 开始发送请求
        // HTTP响应状态码是不是 2xx 开头的
        if (!response.isOk()) {
            // 如果失败了可以进行递归重试
            // 规定一个重试测试
            // 重试3次  退出递归  如果成功 退出递归
            log.error("刷新token失败~~~ ");
            return null;
        }

        // JSON字符串
        /*
        * {
              "access_token": "YWMte3bGuOukEeiTkNP4grL7iwAAAAAAAAAAAAAAAAAAAAGL4CTw6XgR6LaXXVmNX4QCAgMAAAFnKdc-ZgBPGgBFTrLhhyK8woMEI005emtrLJFJV6aoxsZSioSIZkr5kw",
              "expires_in": 5184000,   12个小时  11个小时59分50秒
              "application": "8be024f0-e978-11e8-b697-5d598d5f8402"
            }
        * */
        String jsonBody = response.body();
        // 对JSON字符串进行反序列化
        JSONObject jsonObject = JSONUtil.parseObj(jsonBody);
        // 获取响应体中的Token
        String token = jsonObject.getStr("access_token");
        if (StrUtil.isNotEmpty(token)) {
            //将token数据缓存到redis中，缓存时间由expires_in决定
            //提前1小时失效   12小时  11小时58分钟12秒
            long timeout = jsonObject.getLong("expires_in") - 3600;
            // 存储到Redis中
            this.redisTemplate.opsForValue().set(REDIS_KEY, token, timeout, TimeUnit.SECONDS);
            return token;
        }

        return null;
    }
}
